import numpy as np
from sklearn.metrics import roc_auc_score
y_true = np.array([1, 1, 0, 0])
y_scores = np.array([0.9, 0.6, 0.65, 0.2])
print(roc_auc_score(y_true, y_scores))

# y_true = np.array([[1, 0], [1, 0], [0, 1], [0, 1]])
# y_scores = np.array([[0.9, 0.1], [0.6, 0.4], [0.65, 0.35], [0.2, 0.8]])
# print(roc_auc_score(y_true, y_scores))